seed_array=(1 2)
gpu_id=4
seed=0
dataset="VLCS"


network="resnet18"

domain="C"
test_envs=0

for seed in ${seed_array[@]}
do
    CUDA_VISIBLE_DEVICES=${gpu_id} python train.py --dataset ${dataset} --data_dir /home/huyihao/projects/data/${dataset} --opt_type Adam --lr 5e-5 --max_epoch 50 --net ${network} --test_envs ${test_envs} --seed ${seed} --output train_output/${network}/${dataset}/${domain}/${seed}
done

domain="L"
test_envs=1

for seed in ${seed_array[@]}
do
    CUDA_VISIBLE_DEVICES=${gpu_id} python train.py --dataset ${dataset} --data_dir /home/huyihao/projects/data/${dataset} --opt_type Adam --lr 5e-5 --max_epoch 50 --net ${network} --test_envs ${test_envs} --seed ${seed} --output train_output/${network}/${dataset}/${domain}/${seed}
done

domain="S"
test_envs=2

for seed in ${seed_array[@]}
do
    CUDA_VISIBLE_DEVICES=${gpu_id} python train.py --dataset ${dataset} --data_dir /home/huyihao/projects/data/${dataset} --opt_type Adam --lr 5e-5 --max_epoch 50 --net ${network} --test_envs ${test_envs} --seed ${seed} --output train_output/${network}/${dataset}/${domain}/${seed}
done

domain="V"
test_envs=3

for seed in ${seed_array[@]}
do
    CUDA_VISIBLE_DEVICES=${gpu_id} python train.py --dataset ${dataset} --data_dir /home/huyihao/projects/data/${dataset} --opt_type Adam --lr 5e-5 --max_epoch 50 --net ${network} --test_envs ${test_envs} --seed ${seed} --output train_output/${network}/${dataset}/${domain}/${seed}
done



network="resnet50"

domain="C"
test_envs=0

for seed in ${seed_array[@]}
do
    CUDA_VISIBLE_DEVICES=${gpu_id} python train.py --dataset ${dataset} --data_dir /home/huyihao/projects/data/${dataset} --opt_type Adam --lr 5e-5 --max_epoch 50 --net ${network} --test_envs ${test_envs} --seed ${seed} --output train_output/${network}/${dataset}/${domain}/${seed}
done

domain="L"
test_envs=1

for seed in ${seed_array[@]}
do
    CUDA_VISIBLE_DEVICES=${gpu_id} python train.py --dataset ${dataset} --data_dir /home/huyihao/projects/data/${dataset} --opt_type Adam --lr 5e-5 --max_epoch 50 --net ${network} --test_envs ${test_envs} --seed ${seed} --output train_output/${network}/${dataset}/${domain}/${seed}
done

domain="S"
test_envs=2

for seed in ${seed_array[@]}
do
    CUDA_VISIBLE_DEVICES=${gpu_id} python train.py --dataset ${dataset} --data_dir /home/huyihao/projects/data/${dataset} --opt_type Adam --lr 5e-5 --max_epoch 50 --net ${network} --test_envs ${test_envs} --seed ${seed} --output train_output/${network}/${dataset}/${domain}/${seed}
done

domain="V"
test_envs=3

for seed in ${seed_array[@]}
do
    CUDA_VISIBLE_DEVICES=${gpu_id} python train.py --dataset ${dataset} --data_dir /home/huyihao/projects/data/${dataset} --opt_type Adam --lr 5e-5 --max_epoch 50 --net ${network} --test_envs ${test_envs} --seed ${seed} --output train_output/${network}/${dataset}/${domain}/${seed}
done